package com.btl.service.ai.support.rag;

import lombok.Getter;
import lombok.Setter;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;

@Setter
@Getter
@Configuration
@ConfigurationProperties(prefix = "rag")
public class RagProperties {

    @Getter
    @Setter
    public static class Embedding {
        private Model allMpNetBaseV2;
        private Model begSmallZhV15;

        public record Model(String modelPath, String tokenizerPath) {

        }
    }

    @Getter
    @Setter
    public static class Scoring {
        private Model begRerankBase;

        public record Model(String modelPath, String tokenizerPath) {

        }
    }

    @Getter
    @Setter
    public static class Store {
        private Vector qdrant;

        public record Vector(String host, int port) {

        }
    }
    private Embedding embedding;
    private Scoring scoring;
    private Store store;

}
